package com.tanhua.sso.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.sso.mapper.UserMapper;

import com.tanhua.sso.pojo.User;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class UserService {

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    @Autowired
    private HuanXinService huanXinService;

    //用户登录
    public String login(String phone, String code) {

        String redisKey = "CHECK_CODE_" + phone;
        Boolean isNew = false;

        //校验验证码:将用户录入的和redis中存储的进行比较
        String redisData = redisTemplate.opsForValue().get(redisKey);
        if (!StringUtils.equals(code, redisData)) { //验证码错误
            return null;
        }
        //验证码校验成功后需要将其从redis中删除
        redisTemplate.delete(redisKey);

        //判断手机号是否存在
        QueryWrapper<User> wrapper = new QueryWrapper<>();
        wrapper.eq("mobile", phone);
        User user = userMapper.selectOne(wrapper);
        //用户在数据库中不存在,则需要注册
        if (null == user) {
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));
            userMapper.insert(user);
            isNew = true;

            //将新用户注册到环信平台
            huanXinService.register(user.getId());

        }

        //仅测试环信:将数据库中的用户注册到环信
       // huanXinService.register(user.getId());

        //生成token
        Map<String, Object> claims = new HashMap();
        //claims.put("mobile", "1333333333"); 手机号算作敏感数据,不存在payload中
        claims.put("id", user.getId());

        String token = Jwts.builder() //header省略
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(12).toDate()) //设置过期时间，12hrs后该token过期
                .compact();

        try {
            //登陆成功,发送消息告知其他系统
            Map<String, Object> msg = new HashMap();
            msg.put("id", user.getId());
            msg.put("date", System.currentTimeMillis());
            rocketMQTemplate.convertAndSend("tanhua-sso-login", msg);
        } catch (MessagingException e) {
            log.error("发送消息失败!", e);
        }

        return token + "|" + isNew;
    }

    public User queryUserByToken(String token) {
        try {
            // 通过token解析数据
            Map<String, Object> body = Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            User user = new User();
            //{mobile=1333333333, id=2, exp=1605513392}
            user.setId(Long.valueOf(body.get("id").toString()));
            //jwt中没有mobile的数据,需要去数据库查询,但如果每次都查询数据库会影响性能,
            //因此对用户手机号进行缓存并设置对应过期时间,过期时间要与token时间一致;
            //注意:如果用户修改手机号需要同步修改redis缓存中的手机号
            String rediskey = "TANHUA_USER_MOBILE_" + user.getId();
            if (redisTemplate.hasKey(rediskey)) {//缓存中有,直接获取并返回
                String mobile = redisTemplate.opsForValue().get(rediskey);
                user.setMobile(mobile);
            } else {//缓存中没有去数据库中拿
                User user1 = userMapper.selectById(user.getId());
                user.setMobile(user1.getMobile());
                //同时将该手机号存入redis缓存中
                Long timeout = Long.valueOf(body.get("exp").toString()) * 1000 - System.currentTimeMillis(); //jwt中的过期时间exp单位是秒,我们需要计算还剩余多少时间过期
                redisTemplate.opsForValue().set(rediskey, user1.getMobile(), timeout, TimeUnit.MILLISECONDS);
            }
            return user;
        } catch (ExpiredJwtException e) {
            log.info("token已经过期！ token=" + token, e);
        } catch (Exception e) {
            log.error("token不合法！ token=" + token, e);
        }
        return null;
    }


}
